import os
import torch
import warnings

# 屏蔽关于 'TypedStorage' 的警告
warnings.filterwarnings("ignore", message="TypedStorage is deprecated")

FILE_PATH = r'D:\DATE\摘要数据集合集\nlpcc_data\nlpcc_data.json'
OUTPUT_DIR ='../data/nlpcc/'
TRAIN_SAMPLE_PATH = os.path.join(OUTPUT_DIR, 'train.json')  # 训练
VAL_SAMPLE_PATH = os.path.join(OUTPUT_DIR, 'val.json')  # 验证：快速调参，选择超参数
TEST_SAMPLE_PATH = os.path.join(OUTPUT_DIR, 'test.json')  # 测试

BERT_PAD_ID = 0  # 填充项对应的id是0

MAX_CONTENT_LEN = 500  # 文本的最大长度
MAX_SUMMARY_LEN = 52 # 摘要的最大长度
TRAIN_BATCH = 20 # 训练集批次大小
TEST_BATCH = 20 #测试集批次大小

BART_MODEL = '../model/bart_zh'
# MODEL_DIR = '../results/'
MODEL_DIR = 'D:\Program Files (x86)\MODEL'
EPOCH = 1000
STEP = 40
LR = 1e-4

DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'

if __name__ == '__main__':
    print(torch.tensor([1, 2, 3]).to(DEVICE))